#include <iostream>
#include <signal.h>

#include "../comm/httplib.h"
#include "oj_control.hpp"

using namespace httplib;
using namespace ns_control;

static Control* ctrl_ptr = nullptr;

void Recovery(int signo)
{
    ctrl_ptr->RecoveryMachines();
}

int main()
{
    signal(SIGQUIT, Recovery);
    // 用户请求的服务路由功能
    Server svr;
    // ctrl_ptr = new Control();
    Control ctrl;
    ctrl_ptr = &ctrl;

    // 获取所有题目列表
    svr.Get("/all_questions", [&ctrl](const Request& req, Response& resp){
        // resp.set_content("这是所有题目的列表", "text/plain; charset=utf8");

        // 获取网页
        std::string html;
        ctrl.AllQuestions(&html);
        resp.set_content(html, "text/html; charset=utf8");
    });

    // 用户根据题目编号获取题目内容
    // (\d+) 正则表达式, \d 代表匹配数字, + 代表匹配一个或多个
    // R"()" 原始字符串, 保持字符串原貌，即可以不转义
    svr.Get(R"(/question/(\d+))", [&ctrl](const Request& req, Response& resp){
        // 例如 questions/100, matches[1] = 100
        std::string id = req.matches[1];   
        // resp.set_content("这是指定的一道题: " + id, "text/plain; charset=utf8");
        
        std::string html;
        ctrl.OneQuestion(&html, id);
        resp.set_content(html, "text/html; charset=utf8");
    });

    // 用户提交代码，使用判题功能
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request& req, Response& resp){
        std::string id = req.matches[1];
        std::string result_json;   
        ctrl.Judge(id, req.body, &result_json);
        resp.set_content(result_json, "application/json;charset=utf-8");

        // resp.set_content("指定判题题目: " + number, "text/plain; charset=utf8");
    });
    
    svr.set_base_dir("./wwwroot");
    svr.listen("0.0.0.0", 8080);
    return 0;
}